package cn.chencaiju.algorithm;

import java.util.HashSet;
import java.util.Set;

import org.junit.Test;

public class GoatLatin {
    public String toGoatLatin(String S) {
    	Set<String> vowel=new HashSet<String>();
    	vowel.add("a");
    	vowel.add("e");
    	vowel.add("i");
    	vowel.add("o");
    	vowel.add("u");
    	vowel.add("A");
    	vowel.add("E");
    	vowel.add("I");
    	vowel.add("O");
    	vowel.add("U");
        StringBuilder sb=new StringBuilder();
        StringBuilder add=new StringBuilder();
        add.append("maa");
        boolean flag=true;
        char initial = '#';//首字母
        for(int i=0;i<S.length();i++){
        	char temp=S.charAt(i);
//        	System.out.print(temp);
        	if(initial=='#') {
        		initial=temp;
//        		System.out.print(temp);
        		if(!vowel.contains(String.valueOf(temp))) {
//        			System.out.print(temp);
        			flag=false;
        			continue;
        		}
        	}
        	if(temp==' ') {
        		if(!flag) {
        			sb.append(initial);
        		}
        		sb.append(add);
        		sb.append(' ');
        		initial='#';
        		flag=true;
        		continue;
        	}
        	sb.append(temp);
        }
        if(initial!='#'&&!flag)
        	sb.append(initial);
        sb.append(add);
        return sb.toString();
    }
    @Test public void testGoatLatin() {
    	System.out.println(toGoatLatin("only"));
//    	toGoatLatin("I speak Goat Latin");
    }
}
